how to check weather my model is on gpu in pytorch

108

how to check weather my model is on gpu in pytorch -

next(network.parameters()).device

Comments

Submit
0 Comments